../../../../models/demos/wormhole/stable_diffusion/tests/test_unet_2d_condition_model.py